package utils

import (
	"gitee.com/saxon134/go-utils/saData/saHit"
	"strings"
)

// FormatContent 0-未知 10-文字 11-链接 20-文档 21-图片 22-视频 23-DOC
// t为10时，识别是普通文字还是链接
// t为20时，识别是图片、视频、DOC，还是文档
// 其他，原样返回
func FormatContent(t int, str string) int {
	if str == "" {
		return 0
	}

	if t == 20 {
		if strings.HasPrefix(str, "https://") == false && strings.HasPrefix(str, "http://") == false {
			return 0
		}

		var suffix = ""
		{
			var ary = strings.Split(str, ".")
			if len(ary) < 2 {
				return 20
			}

			suffix = ary[len(ary)-1]
		}

		switch suffix {
		case "png", "jpg", "jpeg", "gif", "bmp", "ico":
			return 21
		case "mp4", "mov", "avi", "flv", "wmv", "m4v", "qt", "asf", "mpeg", "mpg", "vob":
			return 22
		case "xlsx", "xls", "docx", "doc", "docm", "dotm", "xlsm", "xltm", "xlam", "pptx", "ppt", "potx", "potm", "vsdx", "vsdm":
			return 23
		case "pdf":
			return 24
		default:
			return 20
		}
	}

	if t == 10 && strings.HasPrefix(str, "https://") || strings.HasPrefix(str, "http://") {
		return 11
	}
	return saHit.Int(t > 0, t, 10)
}
